/*
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
 * in compliance with the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License
 * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 * or implied. See the License for the specific language governing permissions and limitations under
 * the License.
 *
 * Copyright @2008 the original author or authors.
 */
package com.bluestone.core;

import java.awt.Rectangle;
import java.awt.Toolkit;
import java.awt.event.InputEvent;
import java.awt.event.KeyEvent;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.FileInputStream;

import javax.imageio.ImageIO;

import com.bluestone.util.Util;

/**
 * If the target is a coordinate(x,y),then save as image;
 * If the target is "saveAsFile",then save as file;
 * If the target is null,then save the full screen as image.
 * @author <a href="mailto:ming7655@gmail.com">Aaron</a>
 */
public class SaveAsAction extends Action {
	private String saveAsFile = "saveAsFile";

	public SaveAsAction(String type, String target, String value, String delay,
			String offset) {
		super(type, target, value, delay, offset);
	}

	@Override
	public boolean execute() {
		if (!super.execute()){
			return false;
		}
		if (target != null) {
			if (target.indexOf(",") > 0) {
				saveAsImage();
			} else if (saveAsFile.equalsIgnoreCase(target)) {
				saveAsFile();
			}
		} else {
			saveFullScreenAsImage();
		}
		boolean isSaveSuccessed = isSaveSuccessed(value);
		if (!isSaveSuccessed) {
			return false;
		}
		if (delay != null && delay.length() > 0) {
			robot.delay(Util.getDelayTime(Integer.parseInt(delay)));
		} else {
			robot.delay(Util.getDelayTime(Util.SAVEASDELAY));
		}
		return true;
	}

	private boolean isSaveSuccessed(String path) {
		File file = new File(path);
		if (!file.exists()) {
			return false;
		}
		try {
			FileInputStream sb = new FileInputStream(file);
			int length = sb.available();
			if (length <= 0) {
				return false;
			}
		} catch (Exception e) {
			Util.error(e);
			return false;
		}
		return true;
	}

	/**
	 * Save as File
	 * 
	 */
	private void saveAsFile() {
		robot.keyPress(KeyEvent.VK_ALT);
		robot.delay(Util.getDelayTime(Util.DELAY300));
		robot.keyPress(KeyEvent.VK_F);
		robot.delay(Util.getDelayTime(Util.DELAY300));
		robot.keyPress(KeyEvent.VK_A);
		robot.delay(Util.getDelayTime(Util.DELAY300));
		robot.keyRelease(KeyEvent.VK_A);
		robot.delay(Util.getDelayTime(Util.DELAY100));
		robot.keyRelease(KeyEvent.VK_F);
		robot.delay(Util.getDelayTime(Util.DELAY100));
		robot.keyRelease(KeyEvent.VK_ALT);
		robot.delay(Util.getDelayTime(Util.DELAY500));
		saveToPath();
		robot.delay(Util.getDelayTime(Util.DELAY1000));
	}

	/**
	 * Save as Image
	 * 
	 */
	private void saveAsImage() {
		int index = target.indexOf("(");
		String temp = target.substring(0,index);
		int pos = target.indexOf(",");
		int endpos = target.indexOf(")");
		int x = Integer.valueOf(target.substring(index+1, pos)).intValue();
		int y = Integer.valueOf(target.substring((pos + 1),endpos)).intValue();
		
		if(Attributes.IEXPLORE.equalsIgnoreCase(temp)){
			y = y+Util.getCurrentProject().getBarHeight();
		}
		
		robot.mouseMove(x, y);
		robot.mousePress(InputEvent.BUTTON3_MASK);
		robot.delay(Util.getDelayTime(Util.DELAY300));
		robot.mouseRelease(InputEvent.BUTTON3_MASK);

		String browserType = Util.getCurrentProject().getBrowser().getType();
		if(Browser.FIREFOX2.equalsIgnoreCase(browserType)){
			robot.delay(Util.getDelayTime(Util.DELAY1000));
			robot.keyPress(KeyEvent.VK_V);
			robot.delay(Util.getDelayTime(Util.DELAY500));
			robot.keyRelease(KeyEvent.VK_V);
			robot.delay(Util.getDelayTime(Util.DELAY1000));			
		}else{
			robot.delay(Util.getDelayTime(Util.DELAY1000));
			robot.keyPress(KeyEvent.VK_S);
			robot.delay(Util.getDelayTime(Util.DELAY500));
			robot.keyRelease(KeyEvent.VK_S);
			robot.delay(Util.getDelayTime(Util.DELAY1000));
		}

		saveToPath();
	}

	private void saveFullScreenAsImage() {
		BufferedImage fullScreenImage = robot
				.createScreenCapture(new Rectangle(Toolkit.getDefaultToolkit()
						.getScreenSize()));
		try {   
			checkSaveAsPath(value);
			ImageIO.write(fullScreenImage, "JPEG", new File(value));
		} catch (Exception e) {
			Util.error(e);
		}
	}

	private void saveToPath() {
		
		value = new File(value).getAbsolutePath();
		Action inputAction = ActionFactory.createAction(
				ActionFactory.INPUTTEXTACTION, null, value, null, null);
		checkSaveAsPath(value);
		if (inputAction.execute()) {
			Action enterAction = ActionFactory.createAction(
					ActionFactory.INPUTKEYACTION, InputKeyAction.ENTERKEY,
					null, null, null);
			if (checkPathIsExisted(value)) {
				enterAction.execute();
				Action leftAction = ActionFactory.createAction(
						ActionFactory.INPUTKEYACTION, InputKeyAction.LEFTKEY,
						null, null, null);
				leftAction.execute();
				robot.delay(Util.getDelayTime(Util.DELAY100));
				enterAction.execute();
			} else {
				enterAction.execute();
			}
		}
	}

	private boolean checkPathIsExisted(String path) {
		File file = new File(path);
		return file.exists();
	}

	private void checkSaveAsPath(String path) {
		String folder = path.substring(0, path.lastIndexOf("\\"));
		File file = new File(folder);
		if (!file.exists()) {
			file.mkdirs();
		}
	}

	public static void main(String[] args) {
		String path = "C:\\bluestone1\\images\\aaa.gif";
		SaveAsAction action = new SaveAsAction("saveas", null, path, null, null);
		System.out.println(action.execute());

	}

}
